
/* vim: set sw=4 sts=4 tw=80 */

/*
 * Copyright (c) 2010 , David B. Cortarello <dcortarello@gmail.com>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *    This product includes software developed by David B. Cortarello.
 * 4. Neither the name of David B. Cortarello nor the
 *    names of its contributors may be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY DAVID B. CORTARELLO 'AS IS' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL DAVID B. CORTARELLO BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/select.h>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <errno.h>
#include "debug.h"

int readt(int fd, char *buffer, int len, struct timeval *timeout)
{
	fd_set rds;
	int i = 0, c = 0;

	while (1) {
		/* Clean up the poll array */
		FD_ZERO(&rds);
		FD_SET(fd, &rds);

		/* Check for the data */
		if ((i = select(fd+1, &rds, NULL, NULL, timeout)) < 0) {
			derror("%s\n", strerror(errno));
			return -2;
		}
		else if (i == 0)
			/* There's no sockets to read from */
			if (c == 0)
				return 2;
			else
				/* This would be the normal case */
				return c;
		else {

			/* There's sockets to read from. Let's read */
			if ((i = read(fd, buffer+c, len-c)) < 0) {
				/* Oooops, error */
				derror("%s\n", strerror(errno));
				return -2;
			}
			else if (i == 0)
				/* End of connection */
				return 0;
			if (c == len)
				/* We only wanted for len */
				return len;

			c += i;

			/* if we reached here it is because there was data, but less than 
			 * len, maybe the user didn't write everything, or maybe we sent 
			 * less than len */
			if (*(buffer+c-1) == '\n')
				return c;
		}
	}
	return 0;
}

